from common import image_to_base64
from manager import TaskManager

if __name__ == '__main__':
    # 示例1
    task_manager = TaskManager()
    task_manager.image_to_video()

    # # 示例2
    # # 修改成自己的图片
    # local_image_path = '/path/to/image.png'
    # base64 = image_to_base64(local_image_path)
    # image = base64
    # prompt = '烛光闪烁，镜头逐渐拉近，窗外缓缓飘雪'
    # negative_prompt = '低质量、颗粒状、变形、插画'
    # cfg_scale = 0.4
    # duration = 5
    # task_manager.simple_image_to_video(local_image_path=image,
    #                                    prompt=prompt,
    #                                    negative_prompt=negative_prompt,
    #                                    cfg_scale=cfg_scale,
    #                                    duration=duration)

    # # 示例3
    # 更新所有任务状态
    task_manager.update_all_task_status()
